# 1. 支持新用户注册，新用户名和密码注册到字典中
# a. 创建一个字典用于保存用户数据    {用户名1:密码, 用户名2:密码}
# b. 判断用户名在原字典中存在还是不存在(判断用户名是不是字典key的一部分)
#    - 存在: 用户名已存在，不能注册
#    - 不存在: 提醒用户输入密码，将用户信息写入字典
# 2. 支持老用户登陆，用户名和密码正确提示登陆成功
# userdb = {"zs": "123", "ls": "456"}
# a. 提示用户输入用户名和密码
# b. 比较  ---->   username(用户从键盘输入的同户名) password(输入的密码)
#    - 保证用户输入的密码和字典保存的密码相同
#      if password == userdb.get(username):
#      - if "abc" == userdb.get("zs"):     "abc" == "123" False
#      - if "123" == userdb.get("zs1"):    "123" == None  False
#    - 条件满足: 登陆成功  条件不满足: 登陆失败
# 3. 主程序通过循环询问，进行何种操作，根据用户的选择，执行注册或是登陆操作
userdb = {"zs": "123", "ls": "456"}  # 创建一个字典用于保存用户数据
def login():
    username = input("用户名:")
    password = input("密码:")
    if password == userdb.get(username):
        print("登陆成功")
    else:
        print("登陆失败")
def register():
    username = input("用户名:")
    if username in userdb.keys():  # 用户名已存在，不能注册
        print("用户名已存在，不能注册")
    else:  # 不存在: 提醒用户输入密码，将用户信息写入字典
        password = input("密码:")
        userdb[username] = password
def show_menu():
    welcome = "(0)login/(1)register/(2)exit, 请选择: "
    while True:
        choice = input(welcome)  # 用户从键盘输入选择
        if choice not in ["0", "1", "2"]:  # 非法输入
            print("请重新输入!!!")
            continue  # 立刻执行下一次选择
        elif choice == "0":
            login()  # 调用登录的函数
        elif choice == "1":
            register()  # 调用注册的函数
        else:
            print("byebye~")
            break
if __name__ == '__main__':
    show_menu()


